package com.xxjqr.rbac.converter;

import cn.hutool.core.util.ObjectUtil;
import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.core.enums.IEnum;
import com.xxjqr.rbac.enums.BaseEnum;
import org.springframework.core.convert.converter.Converter;

import java.util.HashMap;
import java.util.Map;

public class StringToEnumConverter<T extends BaseEnum> implements Converter<String, BaseEnum> {
    private Map<String, T> enumMap = new HashMap<>();
    private Map<String, T> nameEnumMap = new HashMap<>();

    /**
     * @描述 StringToEnumConverter构造函数，起到缓存的作用，先把对应的code->enum映射关系缓存起来，匹配到了直接获取
     * @码农 丁昌江
     * @日期 2021/5/24 15:16
     */
    public StringToEnumConverter(Class<T> enumType) {
        //把所有枚举对象读取出来
        T[] enums = enumType.getEnumConstants();
        for (T e : enums) {
            //枚举对象的code作为key，枚举对象作为value
            enumMap.put(e.getCode().toString(), e);
            //枚举对象的name作为key，枚举对象作为value
            nameEnumMap.put(e.getName(), e);
        }
    }

    /**
     * @描述 code->enum 核心代码
     * @码农 丁昌江
     * @日期 2021/5/24 15:33
     */
    @Override
    public T convert(String source) {
        if (StrUtil.isEmpty(source)) {
            return null;
        }
        T t = enumMap.get(source);
        if (ObjectUtil.isNull(t)) {
            t = nameEnumMap.get(source);
            if (ObjectUtil.isNull(source)) {
                throw new IllegalArgumentException("无法匹配对应的枚举类型");
            }
        }
        return t;
    }
}
